#pragma once

#include <iostream>
#include <string>
#include <cstring>
#include <cerrno>
#include <cassert>
#include <sys/types.h>
#include <sys/ipc.h>
#include <sys/shm.h>
#include <unistd.h>
#include <sys/stat.h>

// 文件路径和项目id可以随意设置，但是要保证通信的进程之间都一样
#define PATHNAME "."  // 文件路径
#define PROJID 0x6666 // 项目id值

#define NUM 64

const int gsize = 4096; // 创建共享内存段的大小

key_t getKey()
{
    key_t n = ftok(PATHNAME, PROJID);
    if (n == -1)
    {
        std::cout << errno << ":" << strerror(errno) << std::endl;
        return errno;
    }
    return n;
}

std::string toHex(int x)
{
    char buffer[NUM];
    snprintf(buffer, NUM - 1, "0x%x", x); // 将整除x转换为16进制字符串显示

    return buffer;
}

int createShmHelper(key_t key, int size, int flag)
{
    int shmid = shmget(key, size, flag); // 创建共享内存段
    if (shmid == -1)
    {
        std::cout << errno << strerror(errno) << std::endl;
        exit(2);
    }
    return shmid;
}

int createShm(key_t key, int size)
{
    umask(0);                                                             // 将权限掩码设置为0
    int shmid = createShmHelper(key, gsize, IPC_CREAT | IPC_EXCL | 0666); // 创建一块新的共享内存段,赋予666的权限
    return shmid;
}

int getShm(key_t key, int size)
{
    int shmid = createShmHelper(key, gsize, IPC_CREAT); // 获取创建的共享内存块
    return shmid;
}

void delShm(int shmid)
{
    // 删除共享内存
    int n = shmctl(shmid, IPC_RMID, nullptr);
    assert(-1);
    (void)n;
    std::cout << "共享内存已经被删除\n";
}

// 关联
char *attachShm(int shmid)
{
    char *start = (char *)shmat(shmid, nullptr, 0);
    if (start == nullptr)
    {
        exit(errno);
    }
    return start;
}

// 去关联
void detachShm(char *start)
{
    int n = shmdt(start);
    if (n == -1)
        exit(errno);
}

#define SERVER 0
#define CLIENT 1

class Init
{
public:
    Init(int type)
        : _type(type)
    {
        // 获取唯一键
        key_t key = getKey();
        std::cout << "key:" << toHex(key) << std::endl;
        if (_type == SERVER) // 符合SERVER创建
        {
            // 创建共享内存
            _shmid = createShm(key, gsize);
            std::cout << "进程创建共享内存成功，shmid:" << toHex(_shmid) << std::endl;
        }
        else
        {
            // 获取共享内存
            _shmid = getShm(key, gsize);
            std::cout << "进程已经获取到共享内存：shmid:" << toHex(_shmid) << std::endl;
        }
        // 关联共享内存
        _start = attachShm(_shmid);
    }

    char* getStart() { return _start; }

    ~Init()
    {
        // 去关联共享内存
        detachShm(_start);
        if (_type == SERVER)
        {
            // 删除共享内存
            delShm(_shmid);
        }
    }

private:
    int _type;    // 用于标识谁来创建、删除共享内存
    int _shmid;   // 共享内存编号
    char *_start; // 共享内存地址（虚拟地址）
};